import torch

# 检查是否有可用的 GPU
print(f"Number of available GPUs: {torch.cuda.device_count()}")

# 列出所有可用的 GPU
for i in range(torch.cuda.device_count()):
    print(f"GPU {i}: {torch.cuda.get_device_name(i)}")